package com.smallbeef.springboot_demo.dao;

import com.smallbeef.springboot_demo.bean.Department;
import org.springframework.stereotype.Repository;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

/**
 * 部门 dao
 */
@Repository
public class DepartmentDao {

    // 模拟数据库中的数据
    private static Map<Integer, Department> departments = null;
    static{
        departments = new HashMap<>(); // 创建一个部门表
        departments.put(101, new Department(101, "教学部"));
        departments.put(102, new Department(102, "市场部"));
        departments.put(103, new Department(103, "教研部"));
        departments.put(104, new Department(104, "运营部"));
        departments.put(105, new Department(105, "后勤部"));

    }

    // 获得所有的部门信息
    public Collection<Department> getDepartment(){
        return departments.values();
    }

    // 根据 id 获取部门信息
    public Department getDepartmentById(Integer id){
        return departments.get(id);
    }
}
